Numpy array with different data types